package com.wkbb.basic.utils.jvt;

import com.wkbb.basic.utils.FileUtil;
import lombok.Cleanup;
import lombok.extern.slf4j.Slf4j;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Base64;

/**
 * @author Alger Yang
 * @date 2020/1/17 001717:08
 */
@Slf4j
public class JvtBase64Util {

    public static byte[] url2Data(URL url) throws IOException {
        //将图片文件转化为字节数组字符串，并对其进行Base64编码处理
        System.out.println("图片的路径为:" + url.toString());
        //打开链接
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        //设置请求方式为"GET"
        conn.setRequestMethod("GET");
        //超时响应时间为5秒
        conn.setConnectTimeout(5 * 1000);
        //通过输入流获取图片数据
        InputStream inStream = conn.getInputStream();
        //得到图片的二进制数据，以二进制封装得到数据，具有通用性
        @Cleanup ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        //创建一个Buffer字符串
        byte[] buffer = new byte[1024];
        //每次读取的字符串长度，如果为-1，代表全部读取完毕
        int len = 0;
        //使用一个输入流从buffer里把数据读取出来
        while ((len = inStream.read(buffer)) != -1) {
            //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
            outStream.write(buffer, 0, len);
        }
        //关闭输入流
        inStream.close();
        return outStream.toByteArray();
    }

    public static String data2Base64(byte[] data) throws IOException {
        //转码成jpg图片
        Image imageTookit = Toolkit.getDefaultToolkit().createImage(data);
        BufferedImage cutImage = FileUtil.toBufferedImage(imageTookit);
        @Cleanup ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        ImageIO.write(cutImage, "jpg", outputStream);
        data = outputStream.toByteArray();
        //对字节数组Base64编码
//        BASE64Encoder encoder = new BASE64Encoder();
//        String base64 = encoder.encode(data);
        String base64 = Base64.getEncoder().encodeToString(data);
        log.info("将图片转码为jpg格式");
        //返回Base64编码过的字节数组字符串
        return base64;
    }

}
